import os
import onnxruntime as onnxrt
import numpy as np

model_path = "../csharp/testdata/squeezenet.onnx"
sess = onnxrt.InferenceSession(model_path)
input_name = sess.get_inputs()[0].name
input_shape = sess.get_inputs()[0].shape
output_name = sess.get_outputs()[0].name
output_shape = sess.get_outputs()[0].shape

print(input_name, input_shape)
print(output_name, output_shape)
x = np.zeros(input_shape, dtype = np.float32)

res = sess.run([output_name], {input_name: x})
print(res[0].flatten()[:32])

